tensor([[0.6100, 0.6100, 0.6100, 0.6100, 0.6100],
        [0.3900, 0.3900, 0.3900, 0.3900, 0.3900]], grad_fn=<RoundBackward1>)